package com.example.servicebestpractice;

import android.content.Context;
import android.os.AsyncTask;
import android.os.Environment;
import android.util.Log;

import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.RandomAccessFile;

import okhttp3.OkHttpClient;
import okhttp3.Request;
import okhttp3.Response;

public class DownloadTask extends AsyncTask<String,Integer,Integer> {
    public static final int TYPE_SUCCESS = 0;
    public static final int TYPE_FAILED = 1;
    public static final int TYPE_PAUSED = 2;
    public static final int TYPE_CANCELED = 3;


    private DownloadListener listener;
    /**
     * 判断是否取消或者暂停
     */
    private boolean isPaused = false;
    private boolean isCanceled = false;
    // 下载进度
    private int lastProgress;

    /**
     * 然后在DownloadTask的构造函数中要求传入一个刚刚定义的DownloadListener参数。
     *
     * @param listener 会将下载的状态通过这个参数进行回调。
     */
    public DownloadTask(DownloadListener listener) {
        this.listener = listener;
    }

    /**
     * 用于在后台执行具体的下载逻辑
     *
     * @param params
     * @return
     */
    @Override
    protected Integer doInBackground(String... params) {
        InputStream is = null;
        RandomAccessFile savedFile = null;
        File file = null;
        try {
            // 记录已下载的文件长度
            long downloadedLength = 0;
            // 首先从参数中获取到了下载的URL地址
            String downloadUrl = params[0];
            // 根据URL地址解析出了下载的文件名
            String fileName = downloadUrl.substring(downloadUrl.lastIndexOf("/"));
            // 然后指定将文件下载到 Environment.DIRECTORY_DOWNLOADS目录下，也就是SD卡的Download目录
            String directory = Environment.getExternalStoragePublicDirectory(Environment.DIRECTORY_DOWNLOADS).getPath();
            file = new File(directory + fileName);
            // 判断一下Download目录中是不是已经存在要下载的文件了，如果已经存在的话则读取已下载的字节数，这样就可以在后面启用断点续传的功能。
            if (file.exists()) {
                downloadedLength = file.length();
            }
            // 获取待下载文件的总长度
            long contentLength = getContentLength(downloadUrl);
            Log.d("测试", downloadUrl + "，" + contentLength);
            if (contentLength == 0) {
                // 如果文件长度等于0则说明文件有问题，直接返回TYPE_FAILED ，
                return TYPE_FAILED;
            } else if (contentLength == downloadedLength) {
                // 如果文件长度等于已下载文件长度，那么就说明文件已经下载完了，直接返回 TYPE_SUCCESS
                return TYPE_SUCCESS;
            }
            // 发送网络请求
            OkHttpClient client = new OkHttpClient();
            Request request = new Request.Builder()
                    // 断点下载，指定从哪个字节开始下载，
                    // 在请求中添加了一个header，用于告诉服务器我们想要从哪个字节开始下载，因为已下载过的部分就不需要再重新下载了
                    /*  客户端在发送range请求，会在请求头中增加range请求头，
                        常用格式为： Range: bytes=first-end，
                            first，开始数据的索引位置
                            end，结束数据的索引位置
                        例如：
                            Range: bytes=0-499  就是前500个字符
                            Range: bytes=2-10   第3个字符(索引位置为2)～第11个字符（索引位置为10）
                            Range: bytes=0-     如省略第二个参数，即从索引开始位置，到结束位置
                            Range: bytes=-500    如省略地一个参数，即表示最后500个字符
                    */
                    .addHeader("RANGE", "bytes=" + downloadedLength + "-").url(downloadUrl).build();
            Response response = client.newCall(request).execute();
            // 使用Java的文件流方式，不断从网络上读取数据，不断写入到本地，一直到文件全部下载完成为止
            if (response != null) {
                is = response.body().byteStream();
                // 创建从中读取和向其中写入（可选）的随机访问文件流，该文件由 File参数指定
                /*
                mode 参数指定用以打开文件的访问模式。
                    "r" 以只读方式打开。调用结果对象的任何 write 方法都将导致抛出 IOException。
                    "rw" 打开以便读取和写入。如果该文件尚不存在，则尝试创建该文件。
                    "rws" 打开以便读取和写入，对于 "rw"，还要求对文件的内容或元数据的每个更新都同步写入到底层存储设备。
                    "rwd" 打开以便读取和写入，对于 "rw"，还要求对文件内容的每个更新都同步写入到底层存储设备。
                 */
                savedFile = new RandomAccessFile(file, "rw");
                // 跳过已下载的字节
                savedFile.seek(downloadedLength);
                byte[] b = new byte[1024];
                // 当前下载的总进度
                int total = 0;
                int len;
                while ((len = is.read(b)) != -1) {
                    // 判断用户有没有触发暂停或者取消的操作，如果有的话则返回TYPE_PAUSED或TYPE_CANCELED来中断下载
                    if (isCanceled) {
                        return TYPE_CANCELED;
                    } else if (isPaused) {
                        return TYPE_PAUSED;
                    } else {
                        // 没有取消或者暂停下载
                        total += len;
                        savedFile.write(b, 0, len);
                        // 实时计算当前的下载进度，计算已下载的百分比
                        int progress = (int) ((total + downloadedLength) * 100 / contentLength);
                        // 通知UI界面更新进度
                        publishProgress(progress);
                    }
                }
                response.body().close();
                return TYPE_SUCCESS;
            }
        } catch (Exception e) {
            Log.d("测试", e.getMessage());
        } finally {
            try {
                if (is != null) {
                    is.close();
                }
                if (savedFile != null) {
                    savedFile.close();
                }
                if (isCanceled && file != null) {
                    file.delete();
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
        return TYPE_FAILED;
    }

    /**
     * 用于在界面上更新当前的下载进度
     *
     * @param values
     */
    @Override
    protected void onProgressUpdate(Integer... values) {
        // 首先从参数中获取到当前的下载进度，
        int progress = values[0];
        // 然后和上一次的下载进度进行对比，如果有变化的话则调用DownloadListener的onProgress()方法来通知下载进度更新。
        if (progress > lastProgress) {
            listener.onProgress(progress);
            lastProgress = progress;
        }
    }

    /**
     * 用于通知最终的下载结果
     *
     * @param status
     */
    @Override
    protected void onPostExecute(Integer status) {
        /*
        根据参数中传入的下载状态来进行回调。
        下载成功就调用DownloadListener的onSuccess()方法，
        下载失败就调用onFailed()方法，
        暂停下载就调用onPaused()方法，
        取消下载就调用onCanceled()方法。
        */
        switch (status) {
            case TYPE_SUCCESS:
                listener.onSuccess();
                break;
            case TYPE_FAILED:
                listener.onFailed();
                break;
            case TYPE_PAUSED:
                listener.onPaused();
                break;
            case TYPE_CANCELED:
                listener.onCanceled();
                break;
            default:
                break;
        }
    }

    public void pauseDownload() {
        isPaused = true;
    }

    public void cancelDownload() {
        isCanceled = true;
    }

    private long getContentLength(String downloadUrl) throws IOException {
        OkHttpClient client = new OkHttpClient();
        Request request = new Request.Builder().url(downloadUrl).build();
        Response response = client.newCall(request).execute();
        if (response.body() != null && response.isSuccessful()) {
            long contentLength = response.body().contentLength();
            response.body().close();
            return contentLength;
        }
        return 0;
    }
}
